import os


def change_cwd_to_main():
    if os.path.basename(os.getcwd()) in ['tests', 'drafts', 'examples', 'tools', 'banner_tools']:
        from warnings import warn
        warn(f'--- 工作路径错误, 请在项目根路径运行! cwd: {os.getcwd()}')
        os.chdir('..')

    assert os.path.exists('main.py'), '*** `main.py`不存在, 请在项目根路径运行!'

